package thread;

import java.util.PriorityQueue;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.PriorityBlockingQueue;

/**
 * 定时器
 */

//任务，新建一个类
class MyTask implements Comparable<MyTask>{
    //使用Runnable创建线程，是因为Thread提供了一个构造函数，支持我们把Runnable实例传递过去，创建线程
    //Runnable只是一个接口
    //使用Runnable来描述任务
    private Runnable runnable;
    //描述任务的执行时间，采用时间戳
    private long time;
    //after 当前时间后多久开始执行
    public  MyTask(Runnable runnable,long after){
        this.runnable = runnable;
        this.time = System.currentTimeMillis() + after;
    }

    public void run(){
        this.runnable.run();
    }

    public Runnable getRunnable() {
        return runnable;
    }

    public long getTime() {
        return time;
    }

    /**
     * 比较两个任务的大小
     * 按照执行时间去比较
     * 执行的时间越晚，即time越小，任务越小执行的越早，需要放在队首
     *
     * @param o the object to be compared.
     * @return
     */
    @Override
    //比较当前的对象和o
    //结果 = 0 , 当前实例等于o
    //结果 > 0 , 当前实例大于o，早于
    //结果 < 0 , 当前实例晚于o
    public int compareTo(MyTask o) {
        return (int)(this.time - o.getTime());
    }
}

/**
 * 定时器
 */
public class Mytimer {
    BlockingQueue<MyTask> queue = new PriorityBlockingQueue<>();
    Object locker = new Object();//定义一把锁
    public void schedule(MyTask task){
        try {
            queue.put(task);
            synchronized (locker){
                locker.notify();
            }
        }catch(Exception e){
            e.printStackTrace();
        }
    }
    public Mytimer(){
        //取出队首的任务，判断是否需要执行
        Thread thread = new Thread(()->{
            try{
                while(true){
                    MyTask task = queue.take();
                    //如果时间未到，把任务重新放回队列
                    //如果时间到了，就执行
                    if(task.getTime()>System.currentTimeMillis()) {
                        queue.put(task);
                        //等待一定的时间
                        synchronized (locker) {
                            locker.wait(task.getTime() - System.currentTimeMillis());
                        }
                    }else{
                        task.run();
                    }
                }
            }catch(Exception e){
                e.printStackTrace();
            }
        });
        thread.start();
    }
}
